package controller

import (
	"TiktokMall/myGin/gorm/dbLink"
	_ "TiktokMall/myGin/gorm/dbLink"
	"github.com/gin-gonic/gin"
)

func Test(e *gin.Engine) {
	testGroup := e.Group("/test")
	{
		testGroup.GET("/link", func(c *gin.Context) {
			c.JSON(200, gin.H{"message": "gin正常工作"})
		})

		//查看数据库是否可以正常工作
		testGroup.GET("/db", testDb)
	}
}

func testDb(c *gin.Context) {
	db := dbLink.GetDB()

	// 检查是否存在 usertest 表
	if !db.Migrator().HasTable("usertest") {
		// 如果不存在，则创建表
		type UserTest struct {
			ID   uint   `gorm:"primaryKey"`
			Name string `gorm:"size:255"`
		}
		if err := db.Migrator().CreateTable(&UserTest{}); err != nil {
			c.JSON(500, gin.H{"error": "创建表失败"})
			return
		}
	}

	c.JSON(200, gin.H{"message": "数据库连接正常"})
}
